numpy
scipy>=1.10
torch>=2.1
torch_geometric>=2.4
faiss-cpu
tqdm
scikit-learn

